package com.liuyucheng.concurrent.cyclicBarrier;

import java.util.concurrent.BrokenBarrierException;
import java.util.concurrent.CyclicBarrier;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

public class CyclicBarrierExample {
    public static void main(String[] args) {
        // 创建一个CyclicBarrier实例，指定需要等待的线程数量和屏障动作
        CyclicBarrier barrier = new CyclicBarrier(3, () -> {
            System.out.println("所有运动员准备就绪，比赛开始！");
        });

        // 创建线程池执行运动员任务
        ExecutorService executor = Executors.newFixedThreadPool(3);

        // 模拟3个运动员准备并起跑
        for (int i = 1; i <= 3; i++) {
            final int athleteNumber = i;
            executor.execute(() -> {
                try {
                    System.out.println("运动员" + athleteNumber + "正在热身...");
                    Thread.sleep((long) (Math.random() * 5000)); // 模拟热身时间
                    System.out.println("运动员" + athleteNumber + "热身完毕，等待其他运动员...");

                    // 在屏障处等待，所有线程到达后才会继续执行
                    barrier.await();

                    System.out.println("运动员" + athleteNumber + "起跑！");

                    // 模拟跑步过程
                    Thread.sleep((long) (Math.random() * 5000));
                    System.out.println("运动员" + athleteNumber + "到达终点！");

                    // 第二轮比赛 - CyclicBarrier可以重复使用
                    System.out.println("运动员" + athleteNumber + "准备第二轮比赛...");
                    Thread.sleep((long) (Math.random() * 3000));
                    System.out.println("运动员" + athleteNumber + "第二轮热身完毕，等待其他运动员...");
                    barrier.await();
                    System.out.println("运动员" + athleteNumber + "第二轮起跑！");
                } catch (InterruptedException | BrokenBarrierException e) {
                    e.printStackTrace();
                }
            });
        }

        // 关闭线程池
        executor.shutdown();
    }
}